#!/bin/bash

export CUDA_VISIBLE_DEVICES=0
model_name=ParallelModel

# 设置公共参数
common_params=(
  --task_name long_term_forecast
  --is_training 1
  --root_path ./dataset/ETT-small/
  --data_path ETTh1.csv
  --model_id ETTh1_96_96
  --model $model_name
  --data ETTh1
  --features M
  --seq_len 96
  --label_len 48
  --e_layers 2
  --d_layers 1
  --factor 3
  --enc_in 7
  --dec_in 7
  --c_out 7
  --des 'Exp_Parallel_CNN'
  --d_model 128
  --d_ff 128
  --itr 1
)

# 不同的预测长度
prediction_lengths=(1 3 6 9 12 48 96 144 336 720)
# 循环执行不同预测长度的训练
for pred_len in "${prediction_lengths[@]}"; do
  echo "Running training with prediction length: $pred_len"
  python -u run.py "${common_params[@]}" --pred_len $pred_len
done

echo "All training jobs completed!"